tf.session in tensorflow 2